//package com.lucifer.cloud.boot.blog.ai.model.classification;
//
//import org.deeplearning4j.nn.multilayer.MultiLayerNetwork;
//import org.deeplearning4j.util.ModelSerializer;
//import org.nd4j.evaluation.classification.Evaluation;
//import org.nd4j.linalg.dataset.api.iterator.DataSetIterator;
//
///**
// * @author lucifer
// * @date 2024/11/28 13:52
// */
//public class EvaluateModel {
//    public static void main(String[] args) throws Exception {
//        // 加载训练好的模型
//        MultiLayerNetwork model = ModelSerializer.restoreMultiLayerNetwork(Constant.writeModelPath);
//
//        // 获取测试数据集
//        DataSetIterator testIterator = ImageDataLoader.getDataSetIterator(Constant.trainImageDataPath);
//
//        // 评估模型
//        Evaluation evaluation = model.evaluate(testIterator);
//        System.out.println(evaluation.stats());
//    }
//}
